import draw
import csv
import pandas as pd
import numpy as np
import matplotlib as mpl
from matplotlib import pyplot as plt

csv_file = pd.read_csv("data-grabber/csv_results/relation.csv")
csv_value = np.float32(csv_file.values[:, 1:])

x_labels = csv_file.keys()[1:]
print(x_labels)
cmap = mpl.colors.ListedColormap(["black", "whitesmoke"])
# bounds = [-1.0, -0.5, 0.5, 1.0]
draw.draw_heatmap(csv_value, x_labels, x_labels, "pictures/relation_heatmap.png", cmap=cmap)
